# 请先安装 OpenAI SDK：`pip3 install openai`

# 导入 OpenAI 库
from openai import OpenAI

# 初始化 OpenAI 客户端
# 使用 Deepseek API，需要替换 base_url 为 Deepseek 的 API 地址
client = OpenAI(api_key="sk-9f35dabd24484ffb8c9adc04c1eda706", base_url="https://api.deepseek.com")

# 创建聊天完成请求
# model: 使用 deepseek-chat 模型
# messages: 包含对话历史的消息列表
# stream: 是否使用流式响应
response = client.chat.completions.create(
    model="deepseek-chat",
    messages=[
        {
            "role": "system", 
            "content": """
            你是信息分析师，请根据用户提供的信息返回JSON数据，JSON数据格式如下：
            {
                'code': '', //股票代码
                'name': '', //股票名称
                'buy_range': '', //买入区间
                'sell_range': '', //卖出区间
                'reason': '', //买入理由
                'theme': '', //题材
                'report_date': '', //研报日期
            }。
            如有多个以数组对象返回
            """
        },  # 系统提示，设定 AI 助手的角色
    ],
    stream=True
)

# 打印 AI 助手的回复内容
for chunk in response:
    print(chunk.choices[0].delta.content, end="", flush=True)